package william.tree;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;

/**
 * @author ZhangShenao
 * @date 2024/3/21
 * @description <a href="https://leetcode.cn/problems/path-sum-ii/description/">...</a>
 */
public class Leetcode113_路径总和2 {
    private class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;

        TreeNode() {
        }

        TreeNode(int val) {
            this.val = val;
        }

        TreeNode(int val, TreeNode left, TreeNode right) {
            this.val = val;
            this.left = left;
            this.right = right;
        }
    }

    /**
     * 声明全局变量保存最终结果
     */
    private List<List<Integer>> result = new ArrayList<>();

    /**
     * 采用递归算法
     * 声明一个全局变量保存最终结果
     * 递归深度优先遍历二叉树,并在递归函数中传入前缀路径和剩余的targetSum
     * 直到访问到叶子节点,判断targetSum是否==0,如果是,则说明找到了一条目标路径.保存在最终结果中
     * 回溯: 在递归处理完成后,需要将当前节点从前缀路径中移除
     */
    public List<List<Integer>> pathSum(TreeNode root, int targetSum) {
        //调用递归函数
        dfs(root, new LinkedList<>(), targetSum);

        //返回结果
        return result;
    }

    /**
     * 递归深度优先遍历
     */
    private void dfs(TreeNode node, LinkedList<Integer> prePath, int targetSum) {
        //递归终止条件
        if (node == null) {
            return;
        }

        //当前节点值保存到前缀路径中,并更新targetSum
        prePath.add(node.val);
        targetSum -= node.val;

        //访问到叶子节点,并且targetSum==0,说明找到了一条目标路径,保存结果
        if (node.left == null && node.right == null && targetSum == 0) {
            result.add(new ArrayList<>(prePath));
        }


        //递归遍历左、右子树
        dfs(node.left, prePath, targetSum);
        dfs(node.right, prePath, targetSum);

        //回溯: 在递归处理完成后,需要将当前节点从前缀路径中移除
        prePath.removeLast();
    }

}
